import mysql.connector

host = "localhost"  # 数据库主机地址
user = "root"  # 数据库用户名
password = "123"  # 数据库密码
database = "z_sproot_series"  # 数据库名称
table = "stock_gnzj"  # 数据库表名




def get_cursor(default_database='z_sproot_series'):
    connect = mysql.connector.connect(
        host=host,  # 数据库主机地址
        user=user,  # 数据库用户名
        password=password,  # 数据库密码
        database=default_database  # 数据库名称
    )
    return connect.cursor()


def insert_db(sql, data, default_database='z_sproot_series'):
    mydb = mysql.connector.connect(
        host=host,  # 数据库主机地址
        user=user,  # 数据库用户名
        password=password,  # 数据库密码
        database=default_database  # 数据库名称
    )

    my_cursor = mydb.cursor()
    my_cursor.execute(sql, data)  # 执行SQL插入语句
    print("正在插入数据:", data)  # 输出正在插入的数据
    # 提交更改并关闭数据库连接
    mydb.commit()  # 提交更改
    my_cursor.close()  # 关闭游标对象
    mydb.close()  # 关闭数据库连接


# def insert_mysql(host, user, password, database, table, stat_date,data):
#         # 连接数据库
#     mydb = mysql.connector.connect(
#         host=host,  # 数据库主机地址
#         user=user,  # 数据库用户名
#         password=password,  # 数据库密码
#         database=database  # 数据库名称
#     )
#     # 创建游标对象
#     mycursor = mydb.cursor()
#     sql = f"INSERT INTO {table} (stat_date,seq,hangye,hyzs,hyzs_zdf,lrzj,lczj,zj_ze,gsjs,lzg,lzg_zdf,lzg_dqj) VALUES ('{stat_date}',%s, %s, %s, %s,%s, %s, %s, %s,%s,%s, %s)"  # SQL插入语句
#     # sql = f"INSERT INTO {table} (stat_date,seq,hangye,hangye_code,hyzs,hyzs_zdf,lrzj,lczj,zjze,gsjs,lzg,lzg_code,lzg_zdf,lzg_dqj) VALUES ('{stat_date}',%s, %s, %s, %s,%s, %s, %s, %s,%s,%s, %s, %s, %s)"  # SQL插入语句
#     # sql = "INSERT INTO " + table + "(stat_date,stock_code,stock_name,rzmr,rzye,rqmc,rqyl,rqye,rzrqye) VALUES (" + ss + ",%s, %s, %s, %s,%s, %s, %s, %s)"  # SQL插入语句
#     print(sql)
#     val = data  # 插入的数据
#     print(val)
#     mycursor.execute(sql, val)  # 执行SQL插入语句
#     print("正在插入数据:", val)  # 输出正在插入的数据
#     #提交更改并关闭数据库连接
#     mydb.commit()  # 提交更改
#     mycursor.close()  # 关闭游标对象
#     mydb.close()  # 关闭数据库连接